﻿#ifndef __MAP_COLOR_MODEL_H__
#define __MAP_COLOR_MODEL_H__
#include "JsColorModel.h"
using namespace darknet;

class MapColorModel : public JsColorModel
{
public:
    MapColorModel(const std::string &map_name = "map") : JsColorModel(map_name + ".colors"), m_map_name(map_name)
    {
        m_batch      = 64;
        m_max_epochs = 1000;
    }

protected:
    std::string m_map_name;
};

#endif